34f312414f56fd0dfda4c501d0eeecb0ba19bc9d,core/server/src/test/java/alluxio/worker/file/FileDataManagerTest.java,FileDataManagerTest,persistFileRateLimitingTest,#,143

Before Change


    expectedEvents = Lists.newArrayList("R0.00", "R1.00", "R1.00");
    assertEquals(expectedEvents, mockRateLimiter.readEventsAndClear());

    manager.lockBlocks(fileId, blockIds);
    manager.persistFile(fileId, blockIds);

    expectedEvents = Lists.newArrayList("R1.00", "R1.00", "R1.00");
    assertEquals(expectedEvents, mockRateLimiter.readEventsAndClear());

After Change


    Mockito.when(mUfs.create(Mockito.anyString(), Mockito.any(CreateOptions.class)))
        .thenReturn(outputStream);

    mManager.lockBlocks(fileId, blockIds);
    mManager.persistFile(fileId, blockIds);

    List<String> expectedEvents = Lists.newArrayList("R0.00", "R1.00", "R1.00");
    assertEquals(expectedEvents, mockRateLimiter.readEventsAndClear());